package com.cg.leetcode;

import org.junit.Test;

/**
 * 572.另一棵树的子树
 *
 * @program: LeetCode->LeetCode_572
 * @author: cg
 * @create: 2022-04-21 20:11
 **/
public class LeetCode_572 {

    @Test
    public void test572() {
        TreeNode root = new TreeNode(3);
        root.left = new TreeNode(4);
        root.left.left = new TreeNode(1);
        root.left.right = new TreeNode(2);

        root.right = new TreeNode(5);

        TreeNode subRoot = new TreeNode(4);
        subRoot.left = new TreeNode(1);
        subRoot.right = new TreeNode(2);
        System.out.println(isSubtree(root, subRoot));
    }

    /**
     * 给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在，返回 true ；否则，返回 false 。
     * <p>
     * 二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。
     * <p>
     * 示例 1：
     * 输入：root = [3,4,5,1,2], subRoot = [4,1,2]
     * 输出：true
     * <p>
     * 示例 2：
     * 输入：root = [3,4,5,1,2,null,null,null,null,0], subRoot = [4,1,2]
     * 输出：false
     * <p>
     * 提示：
     * root 树上的节点数量范围是 [1, 2000]
     * subRoot 树上的节点数量范围是 [1, 1000]
     * -10^4 <= root.val <= 10^4
     * -10^4 <= subRoot.val <= 10^4
     *
     * @param root
     * @param subRoot
     * @return
     */
    public boolean isSubtree(TreeNode root, TreeNode subRoot) {
        if (root == null) {
            return false;
        }
        return dfs(root, subRoot) || isSubtree(root.left, subRoot) || isSubtree(root.right, subRoot);
    }

    private boolean dfs(TreeNode root, TreeNode subRoot) {
        if (root == null && subRoot == null) {
            return true;
        }
        if (root == null || subRoot == null || root.val != subRoot.val) {
            return false;
        }
        return dfs(root.left, subRoot.left) && dfs(root.right, subRoot.right);
    }
}
